<?php
/*
 *  $Id: Set.php 7490 2010-03-29 19:53:27Z jwage $
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 * This software consists of voluntary contributions made by many individuals
 * and is licensed under the LGPL. For more information, see
 * <http://www.doctrine-project.org>.
 */

/**
 * Doctrine_Query
 *
 * @package     Doctrine
 * @subpackage  Query
 * @license     http://www.opensource.org/licenses/lgpl-license.php LGPL
 * @link        www.doctrine-project.org
 * @since       1.0
 * @version     $Revision: 7490 $
 * @author      Konsta Vesterinen <kvesteri@cc.hut.fi>
 */
class Doctrine_Query_Set extends Doctrine_Query_Part
{
	public function parse($dql)
	{
		$terms = $this->_tokenizer->sqlExplode($dql, ' ');
		$termsTranslation = array();
		 
		foreach ($terms as $term) {
			$termOriginal = $term;
			 
			// We need to check for agg functions here
			$matches = array();
			$hasAggExpression = $this->_processPossibleAggExpression($term, $matches);

			$lftExpr = (($hasAggExpression) ? $matches[1] . '(' : '');
			$rgtExpr = (($hasAggExpression) ? $matches[3] . ')' : '');

			preg_match_all("/^([a-zA-Z0-9_]+[\.[a-zA-Z0-9_]+]*)(\sAS\s[a-zA-Z0-9_]+)?/i", $term, $m, PREG_SET_ORDER);

			if (isset($m[0])) {
				$processed = array();

				foreach ($m as $piece) {
					$part = $piece[1];
					$e = explode('.', trim($part));

					$fieldName  = array_pop($e);
					$reference  = (count($e) > 0) ? implode('.', $e) : $this->query->getRootAlias();
					$aliasMap   = $this->query->getQueryComponent($reference);

					if ($aliasMap['table']->hasField($fieldName)) {
						$columnName = $aliasMap['table']->getColumnName($fieldName);
						$columnName = $aliasMap['table']->getConnection()->quoteIdentifier($columnName);

						$part = $columnName;
					}

					$processed[] = $part . (isset($piece[2]) ? $piece[2] : '');
				}

				$termsTranslation[$termOriginal] = $lftExpr . implode(' ', $processed) . $rgtExpr;
			}
		}

		return strtr($dql, $termsTranslation);
	}


	protected function _processPossibleAggExpression(& $expr, & $matches = array())
	{
		$hasAggExpr = preg_match('/(.*[^\s\(\=])\(([^\)]*)\)(.*)/', $expr, $matches);

		if ($hasAggExpr) {
			$expr = $matches[2];

			// We need to process possible comma separated items
			if (substr(trim($matches[3]), 0, 1) == ',') {
				$xplod = $this->_tokenizer->sqlExplode(trim($matches[3], ' )'), ',');

				$matches[3] = array();

				foreach ($xplod as $part) {
					if ($part != '') {
						$matches[3][] = $this->parseLiteralValue($part);
					}
				}

				$matches[3] = '), ' . implode(', ', $matches[3]);
			}
		}

		return $hasAggExpr;
	}
}